heart python code